package com.atguigu.netty.groupchat;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.text.SimpleDateFormat;

/**
 * @author: 叶子
 * @date: 2022/02/18 11:30
 **/
public class GroupChatServerHandler extends SimpleChannelInboundHandler {
    //    定义一个channel组，管理所有的channel
//    GlobalEventExecutor是全局的事件执行器，是一个单例
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);
    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

    //handlerAdded 表示连接建立，一旦连接，第一个被执行
//    将当前Channel加入到ChannelGroup
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
//        将客户加入聊天的信息推送给其他在线的客户端
//        writeAndFlush会将ChannelGroup种所有的Channel遍历，并发送消息，不需要我们自己遍历
        channelGroup.writeAndFlush(sdf.format(new java.util.Date())+"[客户端]" + channel.remoteAddress() + "加入聊天\n");
        channelGroup.add(channel);
    }

    //  断开连接，将xx客户离开信息推送给当前在线的客户端
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.writeAndFlush(sdf.format(new java.util.Date())+"[客户端]" + channel.remoteAddress() + "离开了");
//        channelGroup会自动remove掉channel，所以这里不需要手动channelGroup.remove()
        System.out.println("channelGroup size"+channelGroup.size());
    }

    //    表示Channel处于活动状态，提示xx上线
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(sdf.format(new java.util.Date())+ctx.channel().remoteAddress() + "上线了~");
    }

    //    表示Channel处于不活动状态，提示xx离线了
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(sdf.format(new java.util.Date())+ctx.channel().remoteAddress() + "离线了~");
    }

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception {
//        获取到当前channel
        Channel channel = ctx.channel();
//        遍历channelGroup，根据不通的情况，回送不同的消息
        channelGroup.forEach(ch ->{
            if (channel!=ch){//不是当前的channel，转发消息
                ch.writeAndFlush(sdf.format(new java.util.Date())+"[客户]"+channel.remoteAddress()+"发送了消息："+msg+"\n");
            }else {
                ch.writeAndFlush(sdf.format(new java.util.Date())+"[自己]发送了消息"+msg+"\n");
            }
        } );
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
//        关闭通道
        ctx.close();
    }
}